from sqlalchemy import create_engine
from sqlalchemy.pool import QueuePool

# ... existing code ...

engine = create_engine(
    SQLALCHEMY_DATABASE_URL,
    pool_size=5,  # 连接池大小
    max_overflow=10,  # 超过pool_size后最多可以创建的连接数
    pool_timeout=30,  # 连接池中没有可用连接的等待时间
    pool_recycle=3600,  # 连接重置周期
    pool_pre_ping=True  # 每次连接前ping一下数据库，确保连接有效
)

# ... existing code ... 